/**
 * Copyright (c) 2013-Now http://jeesite.com All rights reserved.
 */
package com.jeesite.modules.sys.web;

import com.jeesite.common.collect.ListUtils;
import com.jeesite.common.collect.MapUtils;
import com.jeesite.common.config.Global;
import com.jeesite.common.lang.StringUtils;
import com.jeesite.common.web.BaseController;
import com.jeesite.modules.custom.service.CodeRuleService;
import com.jeesite.modules.sys.entity.Company;
import com.jeesite.modules.sys.entity.EmpUser;
import com.jeesite.modules.sys.entity.Office;
import com.jeesite.modules.sys.service.EmpUserService;
import com.jeesite.modules.sys.service.OfficeService;
import com.jeesite.modules.sys.utils.UserUtils;
import com.jeesite.modules.sys.web.user.EmpUserController;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.stereotype.Controller;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.ui.Model;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

import java.util.List;
import java.util.Map;

/**
 * 部门Controller
 *
 * @author ThinkGem
 * @version 2014-8-19
 */
@Controller
@RequestMapping(value = "${adminPath}/sys/office")
@ConditionalOnProperty(name = "web.core.enabled", havingValue = "true", matchIfMissing = true)
public class OfficeController extends BaseController {

    @Autowired
    private OfficeService officeService;
    @Autowired
    private EmpUserController empUserController;
    @Autowired
    private CodeRuleService codeRuleService;
    @Autowired
    private EmpUserService empUserService;

    /**
     * 获取部门
     */
    @ModelAttribute
    public Office get(String officeCode, boolean isNewRecord) {
        return officeService.get(officeCode, isNewRecord);
    }

    /**
     * 部门管理主页面
     */
    @RequiresPermissions("sys:office:view")
    @RequestMapping(value = "index")
    public String index(Office office, Model model) {
        model.addAttribute("office", office);
        return "modules/sys/officeIndex";
    }

    /**
     * 部门列表
     *
     * @param office
     */
    @RequiresPermissions("sys:office:view")
    @RequestMapping(value = "list")
    public String list(Office office, Model model) {
        model.addAttribute("listUser", empUserService.findList(new EmpUser()));
        model.addAttribute("office", office);
        return "modules/sys/officeList";
    }

    /**
     * 查询部门数据
     *
     * @param office
     */
    @RequiresPermissions("sys:office:view")
    @RequestMapping(value = "listData")
    @ResponseBody
    public List < Office > listData(Office office, String ctrlPermi) {
        if (StringUtils.isBlank(office.getParentCode())) {
            office.setParentCode(Office.ROOT_CODE);
        }
        if (StringUtils.isNotBlank(office.getViewCode())
                || StringUtils.isNotBlank(office.getOfficeName())
                || StringUtils.isNotBlank(office.getFullName())
                || StringUtils.isNotBlank(office.getCompany().getCompanyCode())) {
            office.setParentCode(null);
        }
        officeService.addDataScopeFilter(office, ctrlPermi);
        List < Office > list = officeService.findList(office);
        return list;
    }

    /**
     * 查看编辑部门
     *
     * @param office
     */
    @RequiresPermissions("sys:office:view")
    @RequestMapping(value = "form")
    public String form(Office office, Model model) {
        // 创建并初始化下一个节点信息
        office = createNextNode(office);
        if (StringUtils.isNotBlank(office.getParentCode())) {
            office.setParent(officeService.get(office.getParentCode()));
        }
        if (office.getIsNewRecord()) {
            Company company = new Company();
            company.setCompanyCode("");
            office.setCompany(company);
        }
		/*if (office.getIsNewRecord()) {
			office.setTreeSort(30);
			Office where = new Office();
			where.setParentCode(office.getParentCode());
			Office last = officeService.getLastByParentCode(where);
			if (last != null){
				office.setTreeSort(last.getTreeSort() + 30);
				office.setViewCode(IdGen.nextCode(last.getViewCode()));
			}else if (office.getParent() != null){
				office.setViewCode(office.getParent().getViewCode() + "001");
			}
		}*/
        // 查询公司所关联的部门信息
        model.addAttribute("office", office);
        return "modules/sys/officeForm";
    }

    /**
     * 查看编辑部门
     *
     * @param office
     */
    @RequiresPermissions("sys:office:view")
    @RequestMapping(value = "detail")
    public String detail(Office office, Model model) {
        // 创建并初始化下一个节点信息
        office = createNextNode(office);
        if (StringUtils.isNotBlank(office.getParentCode())) {
            office.setParent(officeService.get(office.getParentCode()));
        }
		/*if (office.getIsNewRecord()) {
			office.setTreeSort(30);
			Office where = new Office();
			where.setParentCode(office.getParentCode());
			Office last = officeService.getLastByParentCode(where);
			if (last != null){
				office.setTreeSort(last.getTreeSort() + 30);
				office.setViewCode(IdGen.nextCode(last.getViewCode()));
			}else if (office.getParent() != null){
				office.setViewCode(office.getParent().getViewCode() + "001");
			}
		}*/
        model.addAttribute("office", office);
        return "modules/sys/officeDetail";
    }

    /**
     * 创建并初始化下一个节点信息，如：排序号、默认值
     */
    @RequiresPermissions("sys:office:edit")
    @RequestMapping(value = "createNextNode")
    @ResponseBody
    public Office createNextNode(Office office) {
        if (StringUtils.isNotBlank(office.getParentCode())) {
            office.setParent(officeService.get(office.getParentCode()));
        }
        if (office.getIsNewRecord()) {
			/*Office where = new Office();
			where.setParentCode(office.getParentCode());
			Office last = officeService.getLastByParentCode(where);
			// 获取到下级最后一个节点
			if (last != null){
				office.setTreeSort(last.getTreeSort() + 30);
				office.setViewCode(IdGen.nextCode(last.getViewCode()));
			}else if(office.getParent() != null){
				office.setViewCode(office.getParent().getViewCode() + "001");
			}*/
            office.setViewCode(codeRuleService.getCode("department"));
        }
        // 以下设置表单默认数据
        if (office.getTreeSort() == null) {
            office.setTreeSort(Office.DEFAULT_TREE_SORT);
        }
        return office;
    }

    /**
     * 保存部门
     *
     * @param office
     */
    @RequiresPermissions("sys:office:edit")
    @PostMapping(value = "save")
    @ResponseBody
    @Transactional(readOnly = false)
    public String save(@Validated Office office, String cmd, Model model) {
        officeService.save(office);
        if (office.getIsNewRecord()) {
            codeRuleService.updateCurrentNumber("department");
        }
        return renderResult(Global.TRUE, text("保存部门''{0}''成功", office.getOfficeName()));
    }

    /**
     * 停用部门
     *
     * @param office
     */
    @RequiresPermissions("sys:office:edit")
    @RequestMapping(value = "disable")
    @ResponseBody
    public String disable(Office office) {
        Office where = new Office();
        where.setStatus(Office.STATUS_NORMAL);
        where.setParentCodes("," + office.getId() + ",");
        long count = officeService.findCount(where);
        if (count > 0) {
            return renderResult(Global.FALSE, text("该部门包含未停用的子部门！"));
        }
        office.setStatus(Office.STATUS_DISABLE);
        officeService.updateStatus(office);
        return renderResult(Global.TRUE, text("停用部门''{0}''成功", office.getOfficeName()));
    }

    /**
     * 启用部门
     *
     * @param office
     */
    @RequiresPermissions("sys:office:edit")
    @RequestMapping(value = "enable")
    @ResponseBody
    public String enable(Office office) {
        office.setStatus(Office.STATUS_NORMAL);
        officeService.updateStatus(office);
        return renderResult(Global.TRUE, text("启用部门''{0}''成功", office.getOfficeName()));
    }

    /**
     * 删除部门
     *
     * @param office
     */
    @RequiresPermissions("sys:office:delete")
    @RequestMapping(value = "delete")
    @ResponseBody
    public String delete(Office office) {
        officeService.delete(office);
        return renderResult(Global.TRUE, text("删除部门''{0}''成功", office.getOfficeName()));
    }

    /**
     * 获取部门树结构数据
     *
     * @param excludeCode    排除的ID
     * @param parentCode     上级Code
     * @param isAll          是否显示所有部门（true：不进行权限过滤）
     * @param officeTypes    部门类型（1：省级公司；2：市级公司；3：部门）
     * @param companyCode    仅查询公司下的部门
     * @param isShowCode     是否显示编码（true or 1：显示在左侧；2：显示在右侧；false or null：不显示）
     * @param isShowFullName 是否显示全部门名称
     * @param isLoadUser     是否加载部门下的用户（true 一次性加载；lazy 懒加载，点击再加载）
     * @param postCode       部门下的用户过滤岗位
     * @param roleCode       部门下的用户过滤角色
     * @return
     */
    @RequiresPermissions("user")
    @RequestMapping(value = "treeData")
    @ResponseBody
    public List < Map < String, Object > > treeData(String excludeCode, String parentCode, Boolean isAll,
                                                    String officeTypes, String companyCode, String isShowCode, String isShowFullName,
                                                    String isLoadUser, String postCode, String roleCode, String ctrlPermi) {
        List < Map < String, Object > > mapList = ListUtils.newArrayList();
        Office where = new Office();
        where.setStatus(Office.STATUS_NORMAL);
        where.setCompanyCode(companyCode);
        if (!(isAll != null && isAll) || Global.isStrictMode()) {
            officeService.addDataScopeFilter(where, ctrlPermi);
        }
        // 根据父节点过滤数据
        if (StringUtils.isNotBlank(parentCode)) {
            where.setParentCode(parentCode);
            where.setParentCodes("," + parentCode + ",");
        }
        // 根据部门类型过滤数据
        if (StringUtils.isNotBlank(officeTypes)) {
            where.setOfficeType_in(officeTypes.split(","));
        }
        List < Office > list = officeService.findList(where);
        for (int i = 0; i < list.size(); i++) {
            Office e = list.get(i);
            // 过滤非正常的数据
            if (!Office.STATUS_NORMAL.equals(e.getStatus())) {
                continue;
            }
            // 过滤被排除的编码（包括所有子级）
            if (StringUtils.isNotBlank(excludeCode)) {
                if (e.getId().equals(excludeCode)) {
                    continue;
                }
                if (e.getParentCodes().contains("," + excludeCode + ",")) {
                    continue;
                }
            }
            Map < String, Object > map = MapUtils.newHashMap();
            map.put("id", e.getId());
            map.put("pId", e.getParentCode());
            String name = e.getOfficeName();
            if ("true".equals(isShowFullName) || "1".equals(isShowFullName)) {
                name = e.getFullName();
            }
            map.put("name", StringUtils.getTreeNodeName(isShowCode, e.getViewCode(), name));
            map.put("title", e.getFullName());
            // 如果需要加载用户，则处理用户数据
            if (StringUtils.inString(isLoadUser, "true", "lazy")) {
                map.put("isParent", true);
                // 一次性后台加载用户，若数据量比较大，建议使用懒加载
                if (StringUtils.equals(isLoadUser, "true")) {
                    List < Map < String, Object > > userList =
                            empUserController.treeData("u_", e.getOfficeCode(), e.getOfficeCode(),
                                    companyCode, postCode, roleCode, isAll, isShowCode, ctrlPermi);
                    mapList.addAll(userList);
                }
            }
            mapList.add(map);
        }
        // 懒加载用户，点击叶子节点的时候再去加载部门（懒加载无法回显，数据量大时，建议使用 listselect 实现列表选择用户）
        if (StringUtils.inString(isLoadUser, "lazy") && StringUtils.isNotBlank(parentCode)) {
            List < Map < String, Object > > userList =
                    empUserController.treeData("u_", parentCode, parentCode,
                            companyCode, postCode, roleCode, isAll, isShowCode, ctrlPermi);
            mapList.addAll(userList);
        }
        return mapList;
    }

    @RequiresPermissions("sys:office:edit")
    @RequestMapping(value = "fixTreeData")
    @ResponseBody
    public String fixTreeData() {
        if (!UserUtils.getUser().isAdmin()) {
            return renderResult(Global.FALSE, text("操作失败，只有管理员才能进行修复！"));
        }
        officeService.fixTreeData();
        return renderResult(Global.TRUE, text("数据修复成功"));
    }

}
